package zoo;

import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;

public class Zoo {
    private List<Animal> animals = new ArrayList<>();

    // 添加动物到动物园
    public void addAnimal(Animal animal) {
        this.animals.add(animal);
    }

    // 通过ID删除动物
    public void removeAnimal(String id) {
        Iterator<Animal> iterator = this.animals.iterator();
        while (iterator.hasNext()) {
            Animal animal = iterator.next();
            if (animal.getId().equals(id)) {
                iterator.remove();
                break;
            }
        }
    }

    // 通过ID查找动物
    public Animal findAnimalById(String id) {
        for (Animal animal : this.animals) {
            if (animal.getId().equals(id)) {
                return animal;
            }
        }
        return null;
    }

    // 显示动物园中所有动物
    public void displayAnimals() {
        for (Animal animal : this.animals) {
            System.out.println(animal);
        }
    }

    // 更新动物名字
    public void updateAnimalName(String id, String newName) {
        Animal animal = findAnimalById(id);
        if (animal != null) {
            animal.setName(newName);
        }
    }
}